package com.kaesar.leetcode.LeetCode_051_100;

public class LeetCode_075 {
    public static void sortColors(int[] nums) {
        if (nums.length == 1) {
            return;
        }
        // zeroIndex用来记录从左开始第一个不为红色的位置，twoIndex用来记录从右开始第一个不为蓝色的位置
        int zeroIndex = 0, twoIndex = nums.length - 1;
        for (int i = 0; i <= twoIndex; ) {
            if (nums[i] == 2) {
                // 如果当前的值等于2，则将当前位置的值和twoIndex位置的值进行交换
                nums[i] = nums[twoIndex];
                nums[twoIndex] = 2;
                twoIndex--;
                continue;
            } else if (nums[i] == 1) {
                i++;
            } else if (nums[i] == 0) {
                // 如果当前的值等于0，则将当前位置的值和zeroIndex位置的值进行交换
                nums[i] = nums[zeroIndex];
                nums[zeroIndex] = 0;
                zeroIndex++;
                i++;
            }
        }
    }

    public static void main(String[] args) {
        int[] nums = new int[]{2, 0, 2, 1, 1, 0};
        sortColors(nums);
        for (int num : nums) {
            System.out.print(num + " ");
        }
    }
}
